#encoding:utf8

import redis, logging
from functools import wraps


def chk_redis(method):
    @wraps(method)
    def wrapper(self, *args, **kwads):
        if self._isEnable():
            return 
        return method(self, *args, **kwads)

    return wrapper



class RedisMgr(redis.Redis):

	_redis = None  
	def __init__(self, host, port, password=None):
		logging.info("Init RedisMgr...")
		HOST = host 
		PORT = port
		try:
			if self._redis == None:
				self._redis = redis.Redis(host = HOST, port=PORT, db=0, password=password)
				logging.info("Redis connect to %s:%s success!"%(HOST, PORT))
			else:
				logging.info("Redis connect inited!")
		except:
			logging.error("Redis conn to %s:%s failed!"%(HOST, PORT))
      

	def _isEnable(self):
		if self._redis == None:
			logging.error("Redis not init!")
			return True 
	
	@chk_redis
	def redis(self):
		return self._redis

'''
    def set(self, key, value):
        return self._redis.set(key, value)
 
    #设置 key 对应的值为 string 类型的 value。如果 key 已经存在,返回 0,nx 是 not exist 的意思
    def setnx(self, key, value):
        return self._redis.setnx(key, value)
 
    #设置 key 对应的值为 string 类型的 value,并指定此键值对应的有效期
    def setex(self, key, time, value):
        return self._redis.setex(key, time, value)
 
    #设置指定 key 的 value 值的子字符串
    #setrange name 8 gmail.com
    #其中的 8 是指从下标为 8(包含 8)的字符开始替换
    def setrange(self, key, num, value):
        return self._redis.setrange(key, num, value)
 
    #获取指定 key 的 value 值的子字符串
    def getrange(self, key, start ,end):
        return self._redis.getrange(key, start, end)
 
    #mget(list)
    def get(self, key):
        if isinstance(key, list):
            return self._redis.mget(key)
        else:
            return self._redis.get(key)
 
    #删除
    def remove(self, key):
        return self._redis.delete(key)
 
    #自增
    def incr(self, key, default = 1):
        if (1 == default):
            return self._redis.incr(key)
        else:
            return self._redis.incr(key, default)
 
    #自减
    def decr(self, key, default = 1):
        if (1 == default):
            return self._redis.decr(key)
        else:
            return self._redis.decr(key, default)
 
    #2. hashes 类型及操作
    #根据email获取session信息
    def hget(self, email):
        return self._redis.hget('session', email)
 
    #以email作为唯一标识，增加用户session
    def hset(self, email, content):
        return self._redis.hset('session', email, content)
 
    #获取session哈希表中的所有数据
    def hgetall(self):
        return self._redis.hgetall('session')
 
    #删除hashes
    def hdel(self, name, key = None):
        if(key):
            return self._redis.hdel(name, key)
        return self._redis.hdel(name)
 
    #清空当前db
    def clear(self):
        return self._redis.flushdb()
 
    #3、lists 类型及操作
    #适合做邮件队列
    #在 key 对应 list 的头部添加字符串元素
    def lpush(self, key ,value):
        return self._redis.lpush(key, value)
 
    #从 list 的尾部删除元素,并返回删除元素
    def lpop(self, key):
        return self._redis.plush(key)

'''